package com.qiangesoft.datasourcepro.core;

import com.qiangesoft.datasourcepro.utils.SpringUtil;

/**
 * 数据源上下文
 *
 * @author qiangesoft
 * @date 2024-03-14
 */
public class DataSourceContext {

    /**
     * 线程本地变量：数据源
     */
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    /**
     * 设置数据源的变量
     */
    public static void setDataSource(String dataSourceId) {
        SpringUtil.getBean(DynamicDataSource.class).addDataSource(dataSourceId);
        CONTEXT_HOLDER.set(dataSourceId);
    }

    /**
     * 获得数据源的变量
     */
    public static String getDataSource() {
        return CONTEXT_HOLDER.get();
    }

    /**
     * 清空数据源变量
     */
    public static void removeDataSource() {
        CONTEXT_HOLDER.remove();
    }
}